#!/usr/bin/env python
"""
检查Redis队列状态
"""
import os
import django
import json

# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'keep_project0.settings')
django.setup()

from tools.myredis import mredis

def check_queue_status():
    """检查队列状态"""
    print("=== Redis队列状态检查 ===")
    
    # 检查logo_generation_queue队列
    queue_items = mredis.list_lrange('logo_generation_queue', 0, -1)
    print(f"logo_generation_queue 队列长度: {len(queue_items)}")
    
    if queue_items:
        print("队列中的任务:")
        for i, item in enumerate(queue_items):
            try:
                task_data = json.loads(item)
                print(f"  任务 {i+1}: 用户ID {task_data.get('user_id')}, 性别 {task_data.get('gender')}, 健身水平 {task_data.get('fitness_level')}")
            except:
                print(f"  任务 {i+1}: {item}")
    else:
        print("队列为空")
    
    # 检查用户3的结果
    result_data = mredis.get('logo_result_3')
    if result_data:
        try:
            result = json.loads(result_data)
            print(f"\n用户3的处理结果: {result}")
        except:
            print(f"\n用户3的原始结果: {result_data}")
    else:
        print("\n用户3暂无处理结果")
    
    print("=== 检查完成 ===")

if __name__ == '__main__':
    check_queue_status()
